package com.adee.zingfront;

import java.util.HashMap;
import java.util.Map;
import java.util.Random;

/**
 * 实现一个类，用来存储一个班级的学号。实现以下方法：
 * 1. 添加一个学号
 * 2. 查找一个学号是否存在（时间复杂度要求：O(1)）
 * 3. 删除一个学号
 * 4. 随机返回一个学号
 * 5. 返回一个最小的学号（时间复杂度要求：O(1)）
 *
 * 思路：查找时间复杂度为o(1)，选用哈希结构存储，key为学号。
 * 查找最小学号时间复杂度为o(1)，内部维护一个“最小学号”即可，不需要全部排序。
 */
public class Test1 {
    public static void main(String[] args) {

    }

    // 学号类
    static class StudentNum {
        private Map<Integer,Integer> map = new HashMap<>();

        private int min = -1;

        // 1. 添加一个学号
        public void add(int num) {
            map.put(num, null);
            if (min == -1) {
                min = num;
            } else {
                min = Math.min(min, num);
            }
        }

        // 2. 查找一个学号是否存在（时间复杂度要求：O(1)）
        public boolean exists(int num) {
            return map.containsKey(num);
        }

        // 3. 删除一个学号
        public void remove(int num) {
            map.remove(num);
        }

        // 4. 随机返回一个学号
        public int getRandomNum() {
            Random random = new Random();
            int i = random.nextInt(map.size());
            Integer[] arr = new Integer[map.size()];
            map.keySet().toArray(arr);
            return arr[i];
        }

        // 5. 返回一个最小的学号（时间复杂度要求：O(1)），返回-1表示无任何学号
        public int min() {
            return min;
        }
    }
}
